package com.pg.agent.engine.service.component;


import com.pg.agent.engine.client.model.agent.AIAgent;

/**
 * 规则组件
 *
 * @author huhuaping
 * @date 2023/11/2 11:13
 */
public class RuleComponent extends PromptComponent {
    private String rule;

    public RuleComponent(String rule) {
        super();
        this.rule = rule;
    }

    @Override
    public String getPrompt(AIAgent agent) {
        return String.format("你需要遵守规则：%s\n", rule);
    }
}
